tensor([[0.3200, 0.3200, 0.3200, 0.3200, 0.3200],
        [0.6800, 0.6800, 0.6800, 0.6800, 0.6800]], grad_fn=<RoundBackward1>)